# from text2vec import SentenceModel
# from langchain_ollama import OllamaEmbeddings
# # from transformers import AdamW, get_linear_schedule_with_warmup
import torch
print(torch.cuda.is_available())
# texts = [
#     "娃哈哈",
#     "你好",
#     "你是谁？",
#     "介绍一下自己"
# ]
#
# model = SentenceModel("/Users/shilihua/.cache/modelscope/hub/models/Jerry0/text2vec-base-chinese")
# model.encode(texts)


# embeddings = OllamaEmbeddings(
#     model="quentinz/bge-large-zh-v1.5:latest",
#     base_url="http://10.2.4.31:11434",
# )
# docs = embeddings.embed_documents([
#     "你好",
#     "我的名字是阿海"
# ])
# print(docs)